Searched defs:concat_shape (Results 1 – 11 of 11) sorted by relevance
| /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
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_contiguous.cc | 51 Shape concat_shape = ShapeUtil::MakeShape(element_type, {total_size}); in ReplaceWithContiguousAllReduce() local
|
| D | buffer_assignment_test.cc | 1911 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 2661 Shape concat_shape = in TEST_F() local
2753 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 90}); in TEST_F() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 119 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(… in configure() local
496 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_… in validate() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gru_cell.cc | 42 const RuntimeShape& concat_shape, float* concat, in GruCell()
|
| D | unidirectional_sequence_gru.cc | 62 const RuntimeShape concat_shape = GetTensorShape(concat); in GruImpl() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 468 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_… in validate() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 162 Shape concat_shape; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 848 auto concat_shape = hlo->shape(); in ExchangeHalo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2015 Shape concat_shape({2, num_diags, max_diag_len}); in matchAndRewrite() local
|