HomeSort by relevance Sort by last modified time
    Searched full:mutable_output (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_cwise_ops_common.cc 48 auto out = context->mutable_output(0);
54 out = context->mutable_output(0);
ops_testutil.cc 46 Tensor* output = context_->mutable_output(output_index);
mkl_input_conversion_op.cc 235 << context->mutable_output(0)->shape().DebugString() << " and "
236 << context->mutable_output(1)->shape().DebugString();
470 << context->mutable_output(0)->shape().DebugString() << " and "
471 << context->mutable_output(1)->shape().DebugString();
parse_tensor_test.cc 64 *parse_output = *ctx.mutable_output(0);
random_shuffle_op.cc 86 auto vec = context->mutable_output(0)->vec<T>();
nn_ops_test.cc     [all...]
  /external/tensorflow/tensorflow/contrib/session_bundle/
bundle_shim_test.cc 170 regression_signature->mutable_output()->set_tensor_name("foo-output");
250 foo_regression_signature->mutable_output()->set_tensor_name("foo-output");
256 bar_regression_signature->mutable_output()->set_tensor_name("bar-output");
signature_test.cc 416 signature_.mutable_output()->set_tensor_name(kRegressionsName);
  /external/tensorflow/tensorflow/c/eager/
runtime.cc 301 output_tensors->push_back(Tensor(*context.mutable_output(i)));
  /external/tensorflow/tensorflow/core/framework/
op_kernel.cc 656 DCHECK(mutable_output(index) == nullptr);
724 DCHECK_EQ(mutable_output(index), nullptr);
768 Status OpKernelContext::mutable_output(StringPiece name, Tensor** tensor) { function in class:tensorflow::OpKernelContext
777 *tensor = mutable_output(start);
    [all...]
op_kernel.h 1373 inline Tensor* OpKernelContext::mutable_output(int index) { function in class:tensorflow::OpKernelContext
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc     [all...]
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_launch_op.cc 368 VLOG(3) << ctx->mutable_output(i)->DebugString();

Completed in 290 milliseconds