Searched refs:shape_2 (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | sequential.py | 550 def relax_input_shape(shape_1, shape_2): argument
551 if shape_1 is None or shape_2 is None:
553 if len(shape_1) != len(shape_2):
555 return tuple(None if d1 != d2 else d1 for d1, d2 in zip(shape_1, shape_2))
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_composition_test.py | 164 shape_2 = batch_shape + [k, shape[-1]]
169 shape_2, dtype=dtype)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 136 Shape shape_2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
137 auto layout_2 = shape_2.mutable_layout(); in TEST()
142 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2)); in TEST()
143 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
160 Shape shape_2 = ShapeUtil::MakeShape(PRED, {3, 2}); in TEST() local
161 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 254 const int64 shape_2 = params.shape(2); in SetupGraph() local
266 shape_2, shape_3, data_ptr, data_size); in SetupGraph()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | README.md | 183 shape_2 = TensorShape({3, 8})
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2206 const int64_t shape_2[] = {2, 4, 6}; in TEST_F() local
2207 const int64_t* list[] = {&shape_1[0], &shape_2[0]}; in TEST_F()
2209 const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)}; in TEST_F()
|