Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:feature_index_ (Results 1 - 4 of 4) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.h	72 int64 feature_index_; member in class:xla::gpu::CudnnBatchNormForwardInferenceThunk 100 int64 feature_index_; member in class:xla::gpu::CudnnBatchNormForwardTrainingThunk 135 int64 feature_index_; member in class:xla::gpu::CudnnBatchNormBackwardThunk
cudnn_batchnorm_thunk.cc	92 feature_index_(feature_index), 107 MakeDescriptors(hlo_instruction()->shape(), feature_index_); 146 feature_index_(feature_index), 169 hlo_instruction()->shape().tuple_shapes(0), feature_index_); 226 feature_index_(feature_index), 252 hlo_instruction()->shape().tuple_shapes(0), feature_index_);
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_instruction.h	1476 int64 feature_index_ = -1; member in class:xla::HloInstruction     [all...]
hlo_instruction.cc	143 instruction->feature_index_ = proto.feature_index(); 608 instruction->feature_index_ = feature_index; 625 instruction->feature_index_ = feature_index; 643 instruction->feature_index_ = feature_index;     [all...]

Completed in 81 milliseconds