HomeSort by relevance Sort by last modified time
    Searched defs:GetShape (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/
kernel_util_test.cc 51 std::vector<int> GetShape(TfLiteIntArray* dims) {
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));
test_util.h 68 // m.BuildInterpreter({GetShape(a), GetShape(b)});
136 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; }
  /external/tensorflow/tensorflow/compiler/xla/client/
client.cc 362 StatusOr<Shape> Client::GetShape(const GlobalData& data) {
368 Status s = stub_->GetShape(&request, &response);
computation_builder.cc 224 StatusOr<std::unique_ptr<Shape>> ComputationBuilder::GetShape(
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(rhs)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.h 183 const Shape& GetShape() const {
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 307 std::unique_ptr<Shape> LocalComputationBuilder::GetShape(
309 return builder_.GetShape(operand).ConsumeValueOrDie();
xla_client.py 491 _unwrap_data_handle(operand), self.GetShape(operand),
639 def GetShape(self, operand):
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)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc     [all...]
user_computation.cc     [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
layout_optimizer.cc     [all...]

Completed in 239 milliseconds