| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | single_image_random_dot_stereograms_ops_test.py | 48 shape_1 = sirds_1.get_shape().as_list()
49 self.assertEqual(shape_1, [768, 1024, 1])
52 self.assertAllEqual(shape_1, r_tf_1.shape)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 94 shape_1 = tensor_shape.vector(1)
103 shape_0, shape_1, shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
107 for identity in (shape_0, shape_1):
159 shape_1 = tensor_shape.vector(1)
174 for identity in (shape_0, shape_1):
|
| /external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/ |
| D | large_function_graph.pbtxt | 329 name: "shape_1"
353 input: "shape_1"
369 input: "shape_1"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_composition_test.py | 157 shape_1 = batch_shape + [shape[-2], k]
162 shape_1, dtype=dtype), linear_operator_test_util.random_normal(
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 127 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
128 auto layout_1 = shape_1.mutable_layout(); in TEST()
139 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2)); in TEST()
140 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
156 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
158 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 273 const Shape& shape_1, in HasCompatibleElementTypes() argument
275 return ShapeUtil::SameElementType(shape_0, shape_1) && in HasCompatibleElementTypes()
286 const Shape& shape_1 = instruction->operand(1)->shape(); in HandleRng() local
287 if (!ShapeUtil::IsScalar(shape_0) || !ShapeUtil::IsScalar(shape_1)) { in HandleRng()
293 if (!HasCompatibleElementTypes(shape_0, shape_1, instruction->shape())) { in HandleRng()
|
| D | hlo_verifier.h | 158 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 253 const int64 shape_1 = params.shape(1); in SetupGraph() local
265 node_id + NODE_ID_OFFSET, shape_0, shape_1, in SetupGraph()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 160 auto shape_1 = in CwiseOutputElementCount() local
164 if (shape_1.dim_size() == shape_2.dim_size()) { in CwiseOutputElementCount()
165 for (int i = 0; i < shape_1.dim_size(); i++) { in CwiseOutputElementCount()
167 std::max(shape_1.dim(i).size(), shape_2.dim(i).size())); in CwiseOutputElementCount()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 573 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
594 shape_1,
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | README.md | 182 shape_1 = TensorShape({3, 3})
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2167 const int64_t shape_1[] = {1, 3}; in TEST_F() local
2169 const int64_t* list[] = {&shape_1[0], &shape_2[0]}; in TEST_F()
2171 const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)}; in TEST_F()
|