Searched refs:gather_v2 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 59 void MakeGather(StringPiece name, bool gather_v2, NodeDef* params, in MakeGather() argument
61 if (gather_v2) { in MakeGather()
73 void TestSinglePartition(bool gather_v2, bool include_shared_init, in TestSinglePartition() argument
137 MakeGather("gather", gather_v2, identity_node, input_node, &graph_def); in TestSinglePartition()
285 void TestMultiPartition(bool gather_v2, bool include_shared_init, in TestMultiPartition() argument
378 MakeGather("gather1", gather_v2, identity_node1, input_node, &graph_def); in TestMultiPartition()
379 MakeGather("gather2", gather_v2, identity_node2, input_node, &graph_def); in TestMultiPartition()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 5025 return gen_array_ops.gather_v2(
5032 return gen_array_ops.gather_v2(params, indices, axis, name=name)
5037 def gather_v2(params, function
5052 gather_v2.__doc__ = gather.__doc__
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | py_builtins.py | 619 return array_ops.gather_v2(iterable, order)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_dispatch_test.py | 433 op=array_ops.gather_v2,
|
| D | ragged_dispatch.py | 488 (array_ops.gather_v2, ragged_gather_ops.gather, ['params', 'indices']),
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | data_adapter.py | 1422 cw = array_ops.gather_v2(class_weight_tensor, y_classes)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 4110 // CHECK-LABEL: @gather_v2
4111 func @gather_v2(%arg0: tensor<16x2x3xf32>, %arg1: tensor<16x5xi32>) -> tensor<16x2x5xf32> {
|