| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 69 output.shape().AsProto(instr.mutable_output()->mutable_tensor_shape()); in LogConvAutotuneResults()
70 instr.mutable_output()->set_dtype(output.dtype()); in LogConvAutotuneResults()
92 output.shape().AsProto(instr.mutable_output()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
93 instr.mutable_output()->set_dtype(output.dtype()); in LogFusedConvAutotuneResults()
|
| D | mkl_cwise_ops_common.cc | 48 auto out = context->mutable_output(0); in Compute()
54 out = context->mutable_output(0); in Compute()
|
| D | collective_ops.cc | 99 if (c->mutable_output(0) == nullptr) { in ComputeAsync()
197 if (c->mutable_output(0) == nullptr) { in ComputeAsync()
262 if (c->mutable_output(0) == nullptr) { in ComputeAsync()
331 if (c->mutable_output(0) == nullptr) { in ComputeAsync()
|
| D | mkl_input_conversion_op.cc | 276 << context->mutable_output(kInputIndex_0)->shape().DebugString() in Compute()
278 << context->mutable_output(kInputIndex_1)->shape().DebugString(); in Compute()
|
| D | ops_testutil.cc | 46 Tensor* output = context_->mutable_output(output_index); in GetOutput()
|
| D | random_shuffle_op.cc | 86 auto vec = context->mutable_output(0)->vec<T>(); in Compute()
|
| D | save_restore_tensor.cc | 397 if (dtypes[i] != context->mutable_output(i)->dtype()) { in RestoreTensorsV2()
401 DataTypeString(context->mutable_output(i)->dtype())); in RestoreTensorsV2()
|
| D | collective_nccl_reducer_test.cc | 285 CHECK_EQ(output_tensor_ptr, ctx.mutable_output(0)); in DoReduce()
299 CHECK(tensor_.CopyFrom(*ctx.mutable_output(0), tensor_.shape())); in DoReduce()
|
| D | parse_tensor_test.cc | 64 *parse_output = *ctx.mutable_output(0); in ParseSerializedWithNodeDef()
|
| D | nn_ops_test.cc | 782 testing::ItemsProcessed(context->mutable_output(0)->NumElements() * iters * in BM_LRNFloat()
863 testing::ItemsProcessed(avgpool_context->mutable_output(0)->NumElements() * in BM_AvgPool()
968 testing::ItemsProcessed(avgpool_context->mutable_output(0)->NumElements() * in BM_AvgPoolBk()
1058 testing::ItemsProcessed(maxpool_context->mutable_output(0)->NumElements() * in BM_MaxPool()
1233 testing::ItemsProcessed(relu_context->mutable_output(0)->NumElements() * in BM_ReluFloat()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 72 ASSERT_EQ(expected_shape, ctx.mutable_output(0)->shape()) in TestBitcastOp()
73 << ctx.mutable_output(0)->shape().DebugString(); in TestBitcastOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | get_serialized_resource_op_test.cc | 72 Tensor* output = context_->mutable_output(0); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | bundle_shim_test.cc | 174 regression_signature->mutable_output()->set_tensor_name("foo-output"); in TEST()
254 foo_regression_signature->mutable_output()->set_tensor_name("foo-output"); in TEST()
260 bar_regression_signature->mutable_output()->set_tensor_name("bar-output"); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer_test.cc | 478 CHECK_EQ(output_tensor_ptr, ctx.mutable_output(0)); in DoGather()
491 CHECK(output_tensor_.CopyFrom(*ctx.mutable_output(0), in DoGather()
492 ctx.mutable_output(0)->shape())); in DoGather()
|
| D | base_collective_executor.cc | 239 Tensor* output = ctx->mutable_output(0); in ExecuteAsync()
|
| D | ring_reducer_test.cc | 507 CHECK_EQ(output_tensor_ptr, ctx.mutable_output(0)); in DoReduce()
521 CHECK(tensor_.CopyFrom(*ctx.mutable_output(0), tensor_.shape())); in DoReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 676 CHECK_EQ(output_tensor_ptr, ctx.mutable_output(0)); in DoBroadcast()
692 CHECK(tensor_.CopyFrom(*ctx.mutable_output(0), tensor_.shape())); in DoBroadcast()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 222 ASSERT_EQ(123, ctx.mutable_output(0)->scalar<tensorflow::uint8>()()); in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 1010 Status mutable_output(StringPiece name, Tensor** tensor);
1179 Tensor* mutable_output(int index);
1569 inline Tensor* OpKernelContext::mutable_output(int index) { in mutable_output() function
1655 return ctx_->mutable_output(start_ + i);
|
| D | op_kernel.cc | 732 DCHECK(mutable_output(index) == nullptr); in allocate_output()
800 DCHECK_EQ(mutable_output(index), nullptr); in set_output()
844 Status OpKernelContext::mutable_output(StringPiece name, Tensor** tensor) { in mutable_output() function in tensorflow::OpKernelContext
853 *tensor = mutable_output(start); in mutable_output()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 339 output_tensor = ctx->mutable_output(i); in PopulateOutputs()
380 VLOG(3) << ctx->mutable_output(i)->DeviceSafeDebugString(); in PopulateOutputs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 735 *response.mutable_output() = output; in ExecuteGraphParallel()
903 *result->mutable_output(), in Execute()
931 *result->mutable_output() = execution->result(); in WaitForExecution()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 122 Tensor* output = context->mutable_output(output_index); in GetDatasetFromContext()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 324 output.shape().AsProto(instr.mutable_output()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
325 instr.mutable_output()->set_dtype(output.dtype()); in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 296 outputs->push_back(Tensor(*context.mutable_output(i))); in Run()
|