Home
last modified time | relevance | path

Searched refs:shape_0 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dutil.cc41 std::vector<int> shape_0; in IsBroadcastable() local
43 Get4DShape(input_0, &shape_0); in IsBroadcastable()
45 const int B_0 = shape_0[0]; in IsBroadcastable()
47 const int H_0 = shape_0[1]; in IsBroadcastable()
49 const int W_0 = shape_0[2]; in IsBroadcastable()
51 const int C_0 = shape_0[3]; in IsBroadcastable()
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes_test.py93 shape_0 = tensor_shape.TensorShape([])
103 shape_0, shape_1, shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
107 for identity in (shape_0, shape_1):
158 shape_0 = tensor_shape.TensorShape([])
174 for identity in (shape_0, shape_1):
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py99 shape_0, shape_1 = tf.shape_n([op.inputs[0], op.inputs[1]])
102 shape_0,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc447 bool ShapeVerifier::HasCompatibleElementTypes(const Shape& shape_0, in HasCompatibleElementTypes() argument
450 return ShapeUtil::SameElementType(shape_0, shape_1) && in HasCompatibleElementTypes()
451 (ShapeUtil::SameElementType(shape_0, result_shape) || in HasCompatibleElementTypes()
453 ShapeUtil::SameElementTypeIgnoringFpPrecision(shape_0, in HasCompatibleElementTypes()
460 const Shape& shape_0 = instruction->operand(0)->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()
475 PrimitiveType element_type = shape_0.element_type(); in HandleRng()
Dhlo_verifier.h192 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc95 const BHWC shape_0{input_shape.b, 36, tiles_x * tiles_y, input_shape.c}; in WinogradFromNode() local
99 gpu_info, shape_0, 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/
Dhexagon_control_wrapper.cc252 const int64 shape_0 = params.shape(0); in SetupGraph() local
265 node_id + NODE_ID_OFFSET, shape_0, shape_1, in SetupGraph()
/external/tensorflow/tensorflow/python/ops/
Dnn_grad.py578 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
588 shape_0,
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc1269 auto shape_0 = ops::Shape(s.WithOpName("s"), conv_0); in TEST_F() local
1270 auto i_0 = ops::Identity(s.WithOpName("i"), shape_0); in TEST_F()