| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | util.cc | 42 std::vector<int> shape_1; in IsBroadcastable() local
44 Get4DShape(input_1, &shape_1); in IsBroadcastable()
46 const int B_1 = shape_1[0]; in IsBroadcastable()
48 const int H_1 = shape_1[1]; in IsBroadcastable()
50 const int W_1 = shape_1[2]; in IsBroadcastable()
52 const int C_1 = shape_1[3]; in IsBroadcastable()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 94 shape_1 = tensor_shape.TensorShape([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.TensorShape([1])
174 for identity in (shape_0, shape_1):
|
| /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/core/grappler/costs/graph_properties_testdata/ |
| D | large_function_graph.pbtxt | 338 name: "shape_1"
362 input: "shape_1"
378 input: "shape_1"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_composition_test.py | 163 shape_1 = batch_shape + [shape[-2], k]
168 shape_1, dtype=dtype), linear_operator_test_util.random_normal(
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 99 shape_0, shape_1 = tf.shape_n([op.inputs[0], op.inputs[1]])
111 shape_1,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 130 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
131 auto layout_1 = shape_1.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()
159 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
161 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 96 const BHWC shape_1{input_shape.b, 36, tiles_x * tiles_y, output_shape.c}; in WinogradFromNode() local
105 gpu_info, shape_1, op_def.src_tensors[0].storage_type, in WinogradFromNode()
109 gpu_subgraph->new_tensors = {{shape_0, td_0}, {shape_1, td_1}}; in WinogradFromNode()
|
| /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/compiler/xla/service/ |
| D | hlo_verifier.cc | 448 const Shape& shape_1, in HasCompatibleElementTypes() argument
450 return ShapeUtil::SameElementType(shape_0, shape_1) && in HasCompatibleElementTypes()
461 const Shape& shape_1 = instruction->operand(1)->shape(); in HandleRng() local
462 if (!ShapeUtil::IsScalar(shape_0) || !ShapeUtil::IsScalar(shape_1)) { in HandleRng()
468 if (!HasCompatibleElementTypes(shape_0, shape_1, instruction->shape())) { in HandleRng()
|
| D | hlo_verifier.h | 192 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | README.md | 182 shape_1 = TensorShape({3, 3})
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 578 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
599 shape_1,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 1272 auto shape_1 = ops::Shape(s.WithOpName("s"), conv_1); in TEST_F() local
1273 auto i_1 = ops::Identity(s.WithOpName("i"), shape_1); in TEST_F()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2205 const int64_t shape_1[] = {1, 3}; 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()
|