Searched refs:shape_1 (Results 1 - 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
H A 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/ |
H A 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/python/kernel_tests/linalg/ |
H A D | linear_operator_composition_test.py | 164 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 D | shape_util_test.cc | 157 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 D | hexagon_control_wrapper.cc | 253 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 D | op_level_cost_estimator.cc | 126 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 D | nn_grad.py | 504 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]]) 517 shape_1,
|
/external/tensorflow/tensorflow/c/ |
H A D | c_api_test.cc | 1967 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