Searched refs:constant_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model_test.cc | 37 const TensorShape constant_shape({input_height, input_width}); in CreateTestGraph() local
39 Tensor constant_tensor(DT_FLOAT, constant_shape); in CreateTestGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression_test.cc | 105 TF_ASSERT_OK_AND_ASSIGN(TensorShape constant_shape, in TEST_F()
107 EXPECT_EQ(TensorShape({}), constant_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 222 auto constant_shape = ShapeUtil::MakeTupleShape({other_shape, rhs_shape}); in TEST_F() local
224 HloInstruction::CreateConstant(Literal::CreateFromShape(constant_shape))); in TEST_F()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | 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),
|
| D | wrappers_test.py | 42 [input_shape, constant_shape] = input_shape
53 shape=(constant_shape[-1], self.units),
|