HomeSort by relevance Sort by last modified time
    Searched refs:mutable_input (Results 1 - 25 of 46) sorted by null

1 2

  /external/tensorflow/tensorflow/contrib/input_pipeline/kernels/
input_pipeline_kernels.cc 46 OP_REQUIRES_OK(ctx, ctx->mutable_input("counter", &counter_tensor, true));
  /external/tensorflow/tensorflow/core/kernels/
restore_v2_op_test.cc 216 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[1];
228 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[2];
240 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[3];
252 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[4];
264 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[5];
277 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[6];
289 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[7];
301 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[8];
313 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[9];
325 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[10]
    [all...]
assign_op.h 60 const Tensor& old_lhs = context->mutable_input(0, /* lock_held */ true);
126 Tensor old_unlocked_lhs = context->mutable_input(0, /* lock_held */ false);
scatter_op_test.cc 57 Tensor params_tensor = *mutable_input(0).tensor;
70 Tensor params_tensor = *mutable_input(0).tensor;
88 Tensor params_tensor = *mutable_input(0).tensor;
107 Tensor params_tensor = *mutable_input(0).tensor;
124 Tensor params_tensor = *mutable_input(0).tensor;
140 Tensor params_tensor = *mutable_input(0).tensor;
156 Tensor params_tensor = *mutable_input(0).tensor;
restore_op_test.cc 206 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[1];
218 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[2];
230 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[3];
242 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[4];
254 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[5];
267 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[6];
279 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[7];
291 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[8];
303 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[9];
315 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[10]
    [all...]
scatter_nd_op_test.cc 58 // Tensor params_tensor = *mutable_input(0).tensor;
71 // Tensor params_tensor = *mutable_input(0).tensor;
89 Tensor params_tensor = *mutable_input(0).tensor;
108 Tensor params_tensor = *mutable_input(0).tensor;
138 Tensor params_tensor = *mutable_input(0).tensor;
154 Tensor params_tensor = *mutable_input(0).tensor;
170 Tensor params_tensor = *mutable_input(0).tensor;
training_op_helpers.h 86 *out = ctx->mutable_input(input, lock_held);
count_up_to_op.cc 35 Tensor tensor = context->mutable_input(0, true);
variable_ops.cc 147 Tensor tmpvar = context->mutable_input(0, false);
170 const Tensor& input_tensor = context->mutable_input(0, false);
dense_update_ops.cc 70 Tensor Tparams = context->mutable_input(0, use_exclusive_lock_);
scatter_op.cc 97 Tensor params = c->mutable_input(0, use_exclusive_lock_);
161 Tensor params = c->mutable_input(0, use_exclusive_lock_);
word2vec_kernels.cc 263 Tensor w_in = ctx->mutable_input(0, false);
266 Tensor w_out = ctx->mutable_input(1, false);
  /external/tensorflow/tensorflow/tools/graph_transforms/
obfuscate_names.cc 79 node->mutable_input()->Clear();
90 *(node->mutable_input()->Add()) = new_input_name;
transform_utils_test.cc 447 *(identity_node.mutable_input()->Add()) = original_copy.name();
708 *(mul_node->mutable_input()->Add()) = "add_node1";
709 *(mul_node->mutable_input()->Add()) = "add_node2:0";
710 *(mul_node->mutable_input()->Add()) = "^const_node1:0";
715 *(add_node1->mutable_input()->Add()) = "missing_input1";
716 *(add_node1->mutable_input()->Add()) = "const_node1:0";
717 *(add_node1->mutable_input()->Add()) = "missing_input2";
722 *(add_node2->mutable_input()->Add()) = "missing_input3";
723 *(add_node2->mutable_input()->Add()) = "const_node1:0";
724 *(add_node2->mutable_input()->Add()) = "^const_node2"
    [all...]
backports.cc 45 concat_node.mutable_input()->Clear();
sparsify_gather.cc 209 n->mutable_input()->SwapElements(i, i + 1);
211 n->mutable_input()->RemoveLast();
532 replaced_graph_def.mutable_node(i)->mutable_input()->RemoveLast();
  /external/tensorflow/tensorflow/contrib/session_bundle/
signature_test.cc 48 input_signature->mutable_input()->set_tensor_name("flow");
100 input_signature->mutable_input()->set_tensor_name("flow");
155 input_signature->mutable_input()->set_tensor_name("flow");
207 input_signature->mutable_input()->set_tensor_name("flow");
287 signature_.mutable_input()->set_tensor_name(kInputName);
364 signature.mutable_input()->set_tensor_name("in:0");
379 signature.mutable_input()->set_tensor_name("in:0");
395 signature.mutable_input()->set_tensor_name("in:0");
415 signature_.mutable_input()->set_tensor_name(kInputName);
473 ->mutable_input()
    [all...]
bundle_shim_test.cc 169 regression_signature->mutable_input()->set_tensor_name("foo-input");
196 classification_signature->mutable_input()->set_tensor_name("foo-input");
249 foo_regression_signature->mutable_input()->set_tensor_name("foo-input");
255 bar_regression_signature->mutable_input()->set_tensor_name("bar-input");
302 foo_classification_signature->mutable_input()->set_tensor_name("foo-input");
309 bar_classification_signature->mutable_input()->set_tensor_name("bar-input");
467 inputs_regression_signature->mutable_input()->set_tensor_name("bar-input");
  /external/tensorflow/tensorflow/core/grappler/optimizers/
dependency_optimizer.cc 44 node->mutable_input()->SwapElements(pos, node->input_size() - 1);
45 node->mutable_input()->RemoveLast();
169 fanout->mutable_input()->SwapElements(i, fanout->input_size() - 1);
170 fanout->mutable_input()->RemoveLast();
205 node->mutable_input()->SwapElements(pos, node->input_size() - 1);
206 node->mutable_input()->RemoveLast();
527 target_node->mutable_input()->SwapElements(input_slot,
530 target_node->mutable_input()->RemoveLast();
auto_parallel.cc 145 *apply_gradients_node->mutable_input(gradient_pos[apply_gradients_op]) =
212 *new_node->mutable_input(i) = new_name;
231 *new_node->mutable_input(i) = new_name;
constant_folding.cc 158 node->mutable_input()->SwapElements(i, node->input_size() - 1);
159 node->mutable_input()->RemoveLast();
374 *output->mutable_input(k) = const_name;
548 *output->mutable_input(k) = out[port]->name();
    [all...]
  /external/tensorflow/tensorflow/contrib/framework/kernels/
zero_initializer_op.cc 42 Tensor input = ctx->mutable_input(0, true);
  /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
scatter_add_ndim_op.cc 34 Tensor input_tensor = context->mutable_input(0, false);
  /external/tensorflow/tensorflow/compiler/jit/
graph_to_functiondef.cc 241 *node_def->mutable_input(i) = strings::StrCat("^", normalized);
249 *node_def->mutable_input(i) = iter->second;
  /external/tensorflow/tensorflow/core/grappler/
utils.cc 321 node->mutable_input()->SwapElements(pos, node->input_size() - 1);
322 node->mutable_input()->RemoveLast();

Completed in 781 milliseconds

1 2