Home
last modified time | relevance | path

Searched refs:GetOrCreateAdd (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlrn_ops.cc56 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
143 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
160 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
Dsoftmax_op.cc73 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in Compile()
117 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
131 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
Dbatch_norm_op.cc189 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
202 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
Dl2loss_op.cc42 *ctx->GetOrCreateAdd(accumulation_type), dims); in Compile()
Dbias_ops.cc117 *ctx->GetOrCreateAdd(accumulation_type), reduce_dims); in Compile()
Dscan_ops.cc99 reducer = ctx->GetOrCreateAdd(dtype); in Compile()
Dfake_quantize_ops.cc265 *ctx->GetOrCreateAdd(accumulation_type)); in Compile()
274 *ctx->GetOrCreateAdd(accumulation_type)); in Compile()
Dimage_ops.cc198 *context->GetOrCreateAdd(accumulation_type), in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_context.h87 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
Dxla_context.cc110 const xla::XlaComputation* XlaContext::GetOrCreateAdd(const DataType type) { in GetOrCreateAdd() function in tensorflow::XlaContext
Dxla_op_kernel.h258 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
Dxla_op_kernel.cc582 const xla::XlaComputation* XlaOpKernelContext::GetOrCreateAdd( in GetOrCreateAdd() function in tensorflow::XlaOpKernelContext
584 return xla_context()->GetOrCreateAdd(type); in GetOrCreateAdd()