| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | gru_cell.cc | 42 const RuntimeShape& concat_shape, float* concat, in GruCell() argument
60 &(concat_arrays_data[0]), concat_shape, concat); in GruCell()
63 FullyConnected(fc_params, concat_shape, concat, gate_weight_shape, in GruCell()
75 auto xh = MapAsArrayWithLastDimAsRows(concat, concat_shape); in GruCell()
80 FullyConnected(fc_params, concat_shape, concat, candidate_weight_shape, in GruCell()
|
| D | gru_cell.h | 37 const RuntimeShape& concat_shape, float* concat,
|
| D | unidirectional_sequence_gru.cc | 60 const RuntimeShape concat_shape = GetTensorShape(concat); in GruImpl() local
71 activation_shape, activation_data, concat_shape, concat_data, fc_params, in GruImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_fusion_test.cc | 70 Shape concat_shape = ShapeUtil::MakeShape(F32, {10, 2 * kNumParams}); in TEST_F() local
80 HloInstruction::CreateConcatenate(concat_shape, slice_params, 1)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op.cc | 147 auto concat_shape = concat.shape(); in Compute() local
149 output_shape(j) = concat_shape[j]; in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 164 Shape concat_shape; member
174 pair.concat_shape = ShapeUtil::MakeShape(shape.element_type(), {2}); in SplitShapeIntoHalves()
213 pair.concat_shape = in SplitShapeIntoHalves()
623 shape_pair.concat_shape); in NormalFloatingPointDistribution()
633 shape_pair.concat_shape.dimensions(), in NormalFloatingPointDistribution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 246 TF_ASSIGN_OR_RETURN(Shape concat_shape, ShapeInference::InferConcatOpShape( in MakeConcatHlo()
249 HloInstruction::CreateConcatenate(concat_shape, operands, dimension)); in MakeConcatHlo()
|
| D | buffer_assignment_test.cc | 1875 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
1877 HloInstruction::CreateConcatenate(concat_shape, {rev, neg}, 0)); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 2246 Shape concat_shape = in TEST_F() local
2250 concat_shape, {param0, param1, param2}, 1)); in TEST_F()
2338 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 90}); in TEST_F() local
2391 concat_shape, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 886 auto concat_shape = hlo->shape(); in ExchangeHalo() local
891 concat_shape.set_dimensions(dim, concat_dim_size); in ExchangeHalo()
893 HloInstruction::CreateConcatenate(concat_shape, concat_pieces, dim)); in ExchangeHalo()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1614 Shape concat_shape({2, num_diags, max_diag_len}); in matchAndRewrite() local
1616 loc, RankedTensorType::get(concat_shape, rewriter.getIntegerType(32)), in matchAndRewrite()
|