Searched refs:shape_n (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 89 return array_ops.shape_n([x_i, x, y, y_i]), array_ops.shape_n(
|
| /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/contrib/boosted_trees/python/ops/ |
| D | batch_ops_utils.py | 75 sizes = array_ops.stack(array_ops.shape_n(tensors))[:, 0]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 226 shapes = array_ops.shape_n([
235 shapes = array_ops.shape_n([
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 83 return array_ops.shape_n(inputs)
97 return array_ops.shape_n(inputs)
|
| D | image_ops_impl.py | 2735 shape1, shape2 = array_ops.shape_n([img1, img2])
2901 shape1, shape2 = array_ops.shape_n([img1, img2])
3090 tails = [x[1:] for x in array_ops.shape_n(downscaled)]
|
| D | array_ops.py | 345 def shape_n(input, out_type=dtypes.int32, name=None): function
360 return gen_array_ops.shape_n(input, out_type=out_type, name=name)
|
| D | nn_grad.py | 573 shape_0, shape_1 = array_ops.shape_n([op.inputs[0], op.inputs[1]])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | shape_ops_test.py | 76 tf_ans = array_ops.shape_n([x, x, x])
77 tf_ans_64 = array_ops.shape_n([x, x, x], out_type=dtypes.int64)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 1318 shapen = array_ops.shape_n([conv, conv])
1351 shapen = array_ops.shape_n([conv, conv_reshape])
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1072 array_ops.shape_n(partition_ids_list), [len(partition_ids_list)])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 924 name: "shape_n"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2020 name: "shape_n"
|