Searched refs:index_vector (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 111 HloInstruction* index_vector, const GatherDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace() argument
113 HloComputation* computation = index_vector->parent(); in ExpandIndexVectorIntoOperandSpace()
114 const Shape& index_shape = index_vector->shape(); in ExpandIndexVectorIntoOperandSpace()
136 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()
168 HloInstruction* index_vector; in GatherLoopBody() local
174 index_vector, in GatherLoopBody()
191 TF_ASSIGN_OR_RETURN(index_vector, in GatherLoopBody()
197 ExpandIndexVectorIntoOperandSpace(index_vector, dim_numbers, in GatherLoopBody()
|
| D | scatter_expander.cc | 133 HloInstruction* index_vector, const ScatterDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace() argument
135 HloComputation* computation = index_vector->parent(); in ExpandIndexVectorIntoOperandSpace()
136 const Shape& index_shape = index_vector->shape(); in ExpandIndexVectorIntoOperandSpace()
159 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()
238 HloInstruction* index_vector; in ScatterLoopBody() local
241 index_vector, in ScatterLoopBody()
253 TF_ASSIGN_OR_RETURN(index_vector, in ScatterLoopBody()
258 ExpandIndexVectorIntoOperandSpace(index_vector, dim_numbers, in ScatterLoopBody()
|
| /external/webrtc/webrtc/common_audio/vad/ |
| D | vad_core.h | 42 int16_t index_vector[16 * kNumChannels]; member
|
| D | vad_sp.c | 72 int16_t* age = &self->index_vector[offset]; in WebRtcVad_FindMinimum()
|
| D | vad_core.c | 513 self->index_vector[i] = 0; in WebRtcVad_InitCore()
|