Home
last modified time | relevance | path

Searched refs:mutable_output (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/executorch/runtime/executor/test/
H A Dmethod_test.cpp159 method->mutable_output(0); in TEST_F()
160 method->mutable_output(num_outputs - 1); in TEST_F()
163 ET_EXPECT_DEATH(method->mutable_output(num_outputs), ""); in TEST_F()
164 ET_EXPECT_DEATH(method->mutable_output(num_outputs + 1), ""); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_cwise_ops_common.cc47 auto out = context->mutable_output(0); in Compute()
53 out = context->mutable_output(0); in Compute()
H A Dmkl_input_conversion_op.cc268 << context->mutable_output(kInputIndex_0)->shape().DebugString() in Compute()
270 << context->mutable_output(kInputIndex_1)->shape().DebugString(); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc146 if (op_context.mutable_output(i) == nullptr) { in EvaluateOperation()
152 ConvertTensor(*(op_context.mutable_output(i)), builder); in EvaluateOperation()
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/
H A Dbitcast_op_test.cc71 ASSERT_EQ(expected_shape, ctx.mutable_output(0)->shape()) in TestBitcastOp()
72 << ctx.mutable_output(0)->shape().DebugString(); in TestBitcastOp()
H A Dsummary_op_test.cc89 &summary, ctx.mutable_output(0)->scalar<tstring>()())); in TestScalarSummaryOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dgpu_utils.cc154 *instr.mutable_output() = output_desc.ToProto(element_type); in LogConvAutotuneResults()
197 *instr.mutable_output() = output_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
H A Dsave_restore_tensor.cc439 if (restore_op.dtype != context->mutable_output(restore_op.idx)->dtype()) { in RestoreTensorsV2()
443 DataTypeString(context->mutable_output(restore_op.idx)->dtype())); in RestoreTensorsV2()
H A Dnn_ops_test.cc821 state.SetItemsProcessed(context->mutable_output(0)->NumElements() * in BM_LRNFloat()
901 state.SetItemsProcessed(avgpool_context->mutable_output(0)->NumElements() * in BM_AvgPool()
1004 state.SetItemsProcessed(avgpool_context->mutable_output(0)->NumElements() * in BM_AvgPoolBk()
1095 state.SetItemsProcessed(maxpool_context->mutable_output(0)->NumElements() * in BM_MaxPool()
1276 state.SetItemsProcessed(relu_context->mutable_output(0)->NumElements() * in BM_ReluFloat()
1348 state.SetItemsProcessed(softplus_context->mutable_output(0)->NumElements() * in BM_SoftplusFloat()
H A Dcollective_ops.cc191 if (c->mutable_output(0) == nullptr) { in ComputeAsyncImpl()
296 if (c->mutable_output(0) == nullptr) { in ComputeAsyncImpl()
371 if (c->mutable_output(0) == nullptr) { in ComputeAsyncImpl()
448 if (c->mutable_output(0) == nullptr) { in ComputeAsyncImpl()
H A Dshuffle_common.h81 auto vec = context->mutable_output(output_idx)->vec<T>(); in RandomShuffle()
H A Dcollective_nccl_test.cc325 CHECK_EQ(output_tensor_ptr, ctx.mutable_output(0)); in RunReduce()
344 CHECK(output_.CopyFrom(*ctx.mutable_output(0), input_.shape())); in RunReduce()
H A Dparse_tensor_test.cc67 *parse_output = *ctx.mutable_output(0); in ParseSerializedWithNodeDef()
H A Dops_testutil.cc198 Tensor* output = context_->mutable_output(output_index); in GetOutput()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h1013 Status mutable_output(StringPiece name, Tensor** tensor);
1172 Tensor* mutable_output(int index);
1593 inline Tensor* OpKernelContext::mutable_output(int index) { in mutable_output() function
1635 return ctx_->mutable_output(start_ + i);
H A Dop_kernel.cc792 if (mutable_output(index) != nullptr) { in allocate_output()
795 " mutable_output(index) = ", mutable_output(index), in allocate_output()
1020 Status OpKernelContext::mutable_output(StringPiece name, Tensor** tensor) { in mutable_output() function in tensorflow::OpKernelContext
1023 *tensor = mutable_output(index); in mutable_output()
/aosp_15_r20/external/tink/testing/cc/
H A Dprf_set_impl.cc50 auto* output = response->mutable_output(); in KeyIds()
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2020-026.md19 outputs->push_back(Tensor(*context.mutable_output(i)));
/aosp_15_r20/external/executorch/runtime/executor/
H A Dmethod.h243 ET_DEPRECATED EValue& mutable_output(size_t i);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dop_builder.cc221 *layer_->mutable_output()->Add() = tensor_id.ToString(); in AddOutput()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dcholesky_op_gpu.cu.cc206 Tensor* output = context->mutable_output(0); in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dthreadpool_device.cc242 Tensor* output = context->mutable_output(i); in LogOutputs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dservice.cc757 *response.mutable_output() = output; in ExecuteGraphParallel()
943 *result->mutable_output(), in Execute()
971 *result->mutable_output() = execution->result(); in WaitForExecution()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/
H A Ddevice_event_mgr_test.cc418 const int64_t return_bytes = ctx->mutable_output(0)->TotalBytes(); in DoAddChain()
419 se::DeviceMemoryBase gpu_src_ptr(DMAHelper::base(ctx->mutable_output(0)), in DoAddChain()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dbase_unary_ops_test.h68 void* result_ptr_on_device = context_->mutable_output(0)->data(); in RunAndExpectResult()

12