| /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 | 90 shape_1 = tensor_shape.TensorShape([1])
99 shape_0, shape_1, shape_4, shape_1x4, shape_4x1, shape_3x4, shape_4x3):
103 for identity in (shape_0, shape_1):
155 shape_1 = tensor_shape.TensorShape([1])
170 for identity in (shape_0, shape_1):
|
| D | tensor_shape_test.py | 242 shape_1 = tensor_shape.TensorShape([1, 2, 3])
247 trace_type.deserialize(trace_type.serialize(shape_1)), shape_1)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | sequential.py | 545 def relax_input_shape(shape_1, shape_2): argument
546 if shape_1 is None or shape_2 is None:
548 if len(shape_1) != len(shape_2):
550 return tuple(None if d1 != d2 else d1 for d1, d2 in zip(shape_1, shape_2))
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-008.md | 28 shape_1 = tf.constant([4, 122], dtype=tf.int64)
30 shapes = [shape_1, shape_2]
|
| D | tfsa-2021-024.md | 22 shape_1 = tf.constant([442, 514, 514, 515, 606, 347, 943, 61, 2], dtype=tf.int64)
24 shapes = [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/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 95 shape_0, shape_1 = tf.shape_n([op.inputs[0], op.inputs[1]])
107 shape_1,
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_composition_test.py | 178 shape_1 = batch_shape + [shape[-2], k]
205 generate_well_conditioned(shape_1, dtype=dtype),
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 121 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
122 auto layout_1 = shape_1.mutable_layout(); in TEST()
133 EXPECT_FALSE(ShapeUtil::Equal(shape_1, shape_2)); in TEST()
134 EXPECT_TRUE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
150 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
152 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | prepare_lifting.td | 74 class HasEqualElementSize<list<int> shape_1, list<int> shape_2> : Constraint<
76 "llvm::ArrayRef<unsigned>({" # !interleave(shape_1, ", ") # "}),"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 798 const Shape& shape_1, in HasCompatibleElementTypes() argument
800 return ShapeUtil::SameElementType(shape_0, shape_1) && in HasCompatibleElementTypes()
811 const Shape& shape_1 = instruction->operand(1)->shape(); in HandleRng() local
812 if (!ShapeUtil::IsScalar(shape_0) || !ShapeUtil::IsScalar(shape_1)) { in HandleRng()
818 if (!HasCompatibleElementTypes(shape_0, shape_1, instruction->shape())) { in HandleRng()
|
| D | hlo_verifier.h | 297 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 | 572 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
593 shape_1,
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2236 const int64_t shape_1[] = {1, 3}; in TEST_F() local
2238 const int64_t* list[] = {&shape_1[0], &shape_2[0]}; in TEST_F()
2240 const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)}; in TEST_F()
|