Searched refs:layer_norm_slice_shape_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 533 layer_norm_slice_shape_ = {n_cell_}; in Initialize()
543 SliceRankedTensor(&builder_, layer_norm_scale_, layer_norm_slice_shape_, in SetCellLayerNormCoefficients()
544 begin_cell_layer_norm_values, layer_norm_slice_shape_, in SetCellLayerNormCoefficients()
555 &builder_, layer_norm_scale_, layer_norm_slice_shape_, in SetInputLayerNormCoefficients()
556 begin_input_layer_norm_values, layer_norm_slice_shape_, in SetInputLayerNormCoefficients()
564 SliceRankedTensor(&builder_, layer_norm_scale_, layer_norm_slice_shape_, in SetForgetLayerNormCoefficients()
565 begin_forget_layer_norm_values, layer_norm_slice_shape_, in SetForgetLayerNormCoefficients()
573 SliceRankedTensor(&builder_, layer_norm_scale_, layer_norm_slice_shape_, in SetOutputLayerNormCoefficients()
574 begin_output_layer_norm_values, layer_norm_slice_shape_, in SetOutputLayerNormCoefficients()
|
| D | lstm_utils.h | 206 SmallVector<int64_t, 1> layer_norm_slice_shape_; variable
|