Searched refs:GetOrCreateAdd (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | lrn_ops.cc | 56 *ctx->GetOrCreateAdd(accumulation_type), in Compile() 143 *ctx->GetOrCreateAdd(accumulation_type), in Compile() 160 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
|
D | softmax_op.cc | 73 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in Compile() 117 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits() 131 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
|
D | batch_norm_op.cc | 189 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile() 202 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
|
D | l2loss_op.cc | 42 *ctx->GetOrCreateAdd(accumulation_type), dims); in Compile()
|
D | bias_ops.cc | 117 *ctx->GetOrCreateAdd(accumulation_type), reduce_dims); in Compile()
|
D | scan_ops.cc | 99 reducer = ctx->GetOrCreateAdd(dtype); in Compile()
|
D | fake_quantize_ops.cc | 265 *ctx->GetOrCreateAdd(accumulation_type)); in Compile() 274 *ctx->GetOrCreateAdd(accumulation_type)); in Compile()
|
D | image_ops.cc | 198 *context->GetOrCreateAdd(accumulation_type), in Compile()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_context.h | 87 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
|
D | xla_context.cc | 110 const xla::XlaComputation* XlaContext::GetOrCreateAdd(const DataType type) { in GetOrCreateAdd() function in tensorflow::XlaContext
|
D | xla_op_kernel.h | 258 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
|
D | xla_op_kernel.cc | 582 const xla::XlaComputation* XlaOpKernelContext::GetOrCreateAdd( in GetOrCreateAdd() function in tensorflow::XlaOpKernelContext 584 return xla_context()->GetOrCreateAdd(type); in GetOrCreateAdd()
|