Home
last modified time | relevance | path

Searched defs:concat_shape (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLLSTMLayer.cpp119 …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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_fusion_test.cc70 Shape concat_shape = ShapeUtil::MakeShape(F32, {10, 2 * kNumParams}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_reduce_contiguous.cc51 Shape concat_shape = ShapeUtil::MakeShape(element_type, {total_size}); in ReplaceWithContiguousAllReduce() local
H A Dbuffer_assignment_test.cc1911 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
H A Dalgebraic_simplifier_test.cc2661 Shape concat_shape = in TEST_F() local
2753 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 90}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dgru_cell.cc42 const RuntimeShape& concat_shape, float* concat, in GruCell()
H A Dunidirectional_sequence_gru.cc62 const RuntimeShape concat_shape = GetTensorShape(concat); in GruImpl() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNELSTMLayer.cpp468 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_… in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dprng.cc162 Shape concat_shape; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc848 auto concat_shape = hlo->shape(); in ExchangeHalo() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc2015 Shape concat_shape({2, num_diags, max_diag_len}); in matchAndRewrite() local