Searched refs:shape_n (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 452 auto shape_n = ops::ShapeN(s.WithOpName("shape_n"), {recv0, recv1}); in TEST_F() local
459 auto add1 = ops::Add(s, shape, shape_n[0]); in TEST_F()
460 auto add2 = ops::Add(s, shape_n[1], shape_n[1]); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 98 const int64_t shape_n = ShapeUtil::TupleElementCount(shape); in MakePartitionedShape() local
99 subshapes.reserve(shape_n); in MakePartitionedShape()
100 for (int64_t i = 0; i < shape_n; ++i) { in MakePartitionedShape()
120 const int64_t shape_n = ShapeUtil::TupleElementCount(shape); in MakeNonPaddedShapeForGivenPartition() local
121 subshapes.reserve(shape_n); in MakeNonPaddedShapeForGivenPartition()
122 for (int64_t i = 0; i < shape_n; ++i) { in MakeNonPaddedShapeForGivenPartition()
|
| /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]])
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 125 return array_ops.shape_n([x_i, x, y,
126 y_i]), array_ops.shape_n([x_i, x, y, y_i],
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 223 shapes = array_ops.shape_n([
232 shapes = array_ops.shape_n([
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | shape_ops_test.py | 72 tf_ans = array_ops.shape_n([x, x, x])
73 tf_ans_64 = array_ops.shape_n([x, x, x], out_type=dtypes.int64)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 86 return array_ops.shape_n(inputs)
100 return array_ops.shape_n(inputs)
|
| D | image_ops_impl.py | 4119 shape1, shape2 = array_ops.shape_n([img1, img2])
4303 shape1, shape2 = array_ops.shape_n([img1, img2])
4551 tails = [x[1:] for x in array_ops.shape_n(downscaled)]
|
| D | nn_grad.py | 572 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
|
| D | array_ops.py | 702 def shape_n(input, out_type=dtypes.int32, name=None): function
717 return gen_array_ops.shape_n(input, out_type=out_type, name=name)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 1831 shapen = array_ops.shape_n([conv, conv])
1867 shapen = array_ops.shape_n([y, y])
1937 shapen = array_ops.shape_n([conv, conv_reshape])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 1004 name: "shape_n"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2112 name: "shape_n"
|