Searched refs:gather_op (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unpartition_embedding_lookup.cc | 130 for (auto* gather_op : gather_ops) { in Run() local
131 auto* op = GetOpWithOutput(*model, gather_op->inputs[1]); in Run()
132 CHECK(op) << "Source of " << gather_op->inputs[1] << " not found"; in Run()
137 LogName(*op), LogName(*gather_op)); in Run()
148 LogName(*op), LogName(*gather_op)); in Run()
176 for (const auto& gather_op : gather_ops) { in Run() local
177 gather_params_concat_op->inputs.push_back(gather_op->inputs[0]); in Run()
236 for (auto* gather_op : gather_ops) { in Run() local
237 DeleteOpAndArraysIfUnused(model, gather_op); in Run()
|
| D | resolve_gather_attributes.cc | 31 auto* gather_op = model->operators[op_index].get(); in Run() local
32 if (gather_op->type != OperatorType::kGather) in Run()
34 auto* op = static_cast<GatherOperator*>(gather_op); in Run()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_nd_op_test.py | 373 gather_op = array_ops.gather_nd(t_params, t_indices)
376 self.evaluate(gather_op)
379 self.evaluate(gather_op)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1220 tensorflow::NodeDef* gather_op = tensorflow_graph->add_node(); in ConvertGatherOperator() local
1221 gather_op->set_op("GatherV2"); in ConvertGatherOperator()
1222 gather_op->set_name(src_op.outputs[0]); in ConvertGatherOperator()
1223 *gather_op->add_input() = src_op.inputs[0]; in ConvertGatherOperator()
1224 *gather_op->add_input() = src_op.inputs[1]; in ConvertGatherOperator()
1229 *gather_op->add_input() = src_op.inputs[2]; in ConvertGatherOperator()
1234 AvailableArrayName(model, gather_op->name() + "/axis"); in ConvertGatherOperator()
1237 *gather_op->add_input() = gather_axis; in ConvertGatherOperator()
1240 (*gather_op->mutable_attr())["Tindices"].set_type(DT_INT32); in ConvertGatherOperator()
1241 (*gather_op->mutable_attr())["Taxis"].set_type(DT_INT32); in ConvertGatherOperator()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 45 "gather_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_gpu_kernels.cu.h | 743 GatherOp gather_op(extent_x, extent_y, extent_z, false);
747 gatherIterType gather_iter(counting_iter, gather_op);
|
| D | BUILD | 882 ":gather_op",
977 name = "gather_op",
978 prefix = "gather_op",
1617 ":gather_op",
5507 "gather_op.cc",
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 138 tensorflow/core/kernels/gather_op.cc
|