HomeSort by relevance Sort by last modified time
    Searched refs:gather_op (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
resolve_gather_attributes.cc 31 auto* gather_op = model->operators[op_index].get(); local
32 if (gather_op->type != OperatorType::kGather)
34 auto* op = static_cast<GatherOperator*>(gather_op);
unpartition_embedding_lookup.cc 130 for (auto* gather_op : gather_ops) {
131 auto* op = GetOpWithOutput(*model, gather_op->inputs[1]);
132 CHECK(op) << "Source of " << gather_op->inputs[1] << " not found";
137 LogName(*op), LogName(*gather_op));
148 LogName(*op), LogName(*gather_op));
176 for (const auto& gather_op : gather_ops) {
177 gather_params_concat_op->inputs.push_back(gather_op->inputs[0]);
236 for (auto* gather_op : gather_ops) {
237 DeleteOpAndArraysIfUnused(model, gather_op);
  /external/tensorflow/tensorflow/python/kernel_tests/
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/
export_tensorflow.cc 1220 tensorflow::NodeDef* gather_op = tensorflow_graph->add_node(); local
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
reduction_gpu_kernels.cu.h 743 GatherOp gather_op(extent_x, extent_y, extent_z, false);
747 gatherIterType gather_iter(counting_iter, gather_op);
    [all...]

Completed in 74 milliseconds