| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | kernel_util_test.cc | 51 std::vector<int> GetShape(TfLiteIntArray* dims) { function in class:tflite::__anon25939::KernelUtilTest
117 EXPECT_THAT(GetShape(output), ::testing::ElementsAre(1, 2));
124 EXPECT_THAT(GetShape(output), ::testing::ElementsAre(2));
134 EXPECT_THAT(GetShape(output), ::testing::ElementsAre(3, 1, 2));
141 EXPECT_THAT(GetShape(output), ::testing::ElementsAre(1, 2, 3, 4));
|
| H A D | test_util.h | 68 // m.BuildInterpreter({GetShape(a), GetShape(b)});
136 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } function in class:tflite::SingleOpModel
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | ir_array.h | 183 const Shape& GetShape() const { function in class:xla::llvm_ir::IrArray
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | client.cc | 362 StatusOr<Shape> Client::GetShape(const GlobalData& data) { function in class:xla::Client
368 Status s = stub_->GetShape(&request, &response);
|
| H A D | computation_builder.cc | 224 StatusOr<std::unique_ptr<Shape>> ComputationBuilder::GetShape( function in class:xla::ComputationBuilder
258 std::unique_ptr<Shape> actual_shape = GetShape(operand).ConsumeValueOrDie();
267 std::unique_ptr<Shape> lhs_shape = GetShape(lhs).ConsumeValueOrDie();
268 std::unique_ptr<Shape> rhs_shape = GetShape(rhs).ConsumeValueOrDie();
299 StatusOr<std::unique_ptr<Shape>> shape_status = GetShape(operand);
400 StatusOr<std::unique_ptr<Shape>> shape = GetShape(operand);
429 StatusOr<std::unique_ptr<Shape>> shape_or_status = GetShape(operand);
534 StatusOr<std::unique_ptr<Shape>> lhs_shape_or_status = GetShape(lhs);
635 StatusOr<std::unique_ptr<Shape>> lhs_shape_or_status = GetShape(lhs);
640 StatusOr<std::unique_ptr<Shape>> rhs_shape_or_status = GetShape(rh
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | local_computation_builder.cc | 307 std::unique_ptr<Shape> LocalComputationBuilder::GetShape( function in class:xla::swig::LocalComputationBuilder
309 return builder_.GetShape(operand).ConsumeValueOrDie();
|
| H A D | xla_client.py | 491 _unwrap_data_handle(operand), self.GetShape(operand),
639 def GetShape(self, operand): member in class:ComputationBuilder
640 return _wrap_shape(self._client.GetShape(_unwrap_data_handle(operand)))
734 padding, self.GetShape(operand).dimensions(),
909 padding, self.GetShape(operand).dimensions(), window_dimensions,
929 shape = Shape(self.GetShape(mu).np_dtype, dims)
949 shape = Shape(self.GetShape(a).np_dtype, dims)
1033 padding, self.GetShape(lhs).dimensions()[2:],
1034 self.GetShape(rhs).dimensions()[2:], window_strides)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.cc | 1237 user_computation->GetShape(arg->operand()));
1287 tensorflow::Status Service::GetShape(const GetShapeRequest* arg, function in class:xla::Service
1316 computation->GetShape(arg->operand()));
|
| H A D | user_computation.cc | 1467 StatusOr<Shape> UserComputation::GetShape(const ComputationDataHandle& handle) { function in class:xla::UserComputation
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | layout_optimizer.cc | 977 TensorShapeProto GetShape(const string& input_name) const { function in class:tensorflow::grappler::__anon26346::Conv2DProcessor
1027 auto filter_shape = GetShape(node_->input(1));
1028 auto input_shape = GetShape(node_->input(0));
1042 auto filter_shape = GetShape(node_->name());
1043 auto input_shape = GetShape(node_->input(0));
1063 auto filter_shape = GetShape(node_->input(1));
1064 auto input_shape = GetShape(node_->name());
|