HomeSort by relevance Sort by last modified time
    Searched refs:constant_shape (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/tools/benchmark/
benchmark_model_test.cc 37 const TensorShape constant_shape({input_height, input_width});
39 Tensor constant_tensor(DT_FLOAT, constant_shape);
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_expression_test.cc 105 TF_ASSERT_OK_AND_ASSIGN(TensorShape constant_shape,
107 EXPECT_EQ(TensorShape({}), constant_shape);
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 222 auto constant_shape = ShapeUtil::MakeTupleShape({other_shape, rhs_shape}); local
224 HloInstruction::CreateConstant(Literal::CreateFromShape(constant_shape)));
  /external/tensorflow/tensorflow/python/keras/layers/
recurrent_test.py 391 [input_shape, constant_shape] = input_shape
403 shape=(constant_shape[-1], self.units),
521 [input_shape, constant_shape] = input_shape
533 shape=(constant_shape[-1], self.units),
    [all...]
wrappers_test.py 42 [input_shape, constant_shape] = input_shape
53 shape=(constant_shape[-1], self.units),

Completed in 172 milliseconds