OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:output_grad_scale
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc
220
const BufferAllocation::Slice&
output_grad_scale
,
232
output_grad_scale_(
output_grad_scale
),
261
se::DeviceMemory<float>
output_grad_scale
(
275
&
output_grad_scale
, &output_grad_offset);
local
278
void* ptrs[] = {output_grad_data.opaque(),
output_grad_scale
.opaque(),
cudnn_batchnorm_thunk.h
119
const BufferAllocation::Slice&
output_grad_scale
,
ir_emitter_unnested.cc
448
auto
output_grad_scale
= assn.GetUniqueSlice(custom_call, {1}).ValueOrDie();
local
460
/*
output_grad_scale
=*/
output_grad_scale
,
[
all
...]
Completed in 1380 milliseconds