Searched refs:shape_1 (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/
H A Dsingle_image_random_dot_stereograms_ops_test.py48 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/
H A Dcommon_shapes_test.py94 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/python/kernel_tests/linalg/
H A Dlinear_operator_composition_test.py164 shape_1 = batch_shape + [shape[-2], k]
169 shape_1, dtype=dtype), linear_operator_test_util.random_normal(
/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc157 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); local
158 auto layout_1 = shape_1.mutable_layout();
169 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2));
170 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2));
186 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); local
188 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2));
/external/tensorflow/tensorflow/core/kernels/hexagon/
H A Dhexagon_control_wrapper.cc253 const int64 shape_1 = params.shape(1); local
265 node_id + NODE_ID_OFFSET, shape_0, shape_1,
/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc126 auto shape_1 = local
130 if (shape_1.dim_size() == shape_2.dim_size()) {
131 for (int i = 0; i < shape_1.dim_size(); i++) {
133 std::max(shape_1.dim(i).size(), shape_2.dim(i).size()));
/external/tensorflow/tensorflow/python/ops/
H A Dnn_grad.py504 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
517 shape_1,
/external/tensorflow/tensorflow/c/
H A Dc_api_test.cc1967 const int64_t shape_1[] = {1, 3}; local
1969 const int64_t* list[] = {&shape_1[0], &shape_2[0]};
1971 const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)};

Completed in 272 milliseconds