Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/engine/
Dsequential.py545 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/
Dtfsa-2021-008.md29 shape_2 = tf.constant([4, 188], dtype=tf.int64)
30 shapes = [shape_1, shape_2]
Dtfsa-2021-024.md23 shape_2 = tf.zeros([9], dtype=tf.int64)
24 shapes = [shape_1, shape_2]
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_composition_test.py179 shape_2 = batch_shape + [k, shape[-1]]
206 generate_well_conditioned(shape_2, dtype=dtype),
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc127 Shape shape_2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
128 auto layout_2 = shape_2.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()
151 Shape shape_2 = ShapeUtil::MakeShape(PRED, {3, 2}); in TEST() local
152 EXPECT_FALSE(ShapeUtil::Compatible(shape_1, shape_2)); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dprepare_lifting.td74 class HasEqualElementSize<list<int> shape_1, list<int> shape_2> : Constraint<
77 "llvm::ArrayRef<unsigned>({" # !interleave(shape_2, ", ") # "}))">,
/external/tensorflow/tensorflow/python/framework/
Dtensor_shape_test.py243 shape_2 = tensor_shape.TensorShape([None, 2, None])
249 trace_type.deserialize(trace_type.serialize(shape_2)), shape_2)
/external/tensorflow/tensorflow/core/util/sparse/
DREADME.md183 shape_2 = TensorShape({3, 8})
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2237 const int64_t shape_2[] = {2, 4, 6}; 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()