| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | concat_spmd_expander.cc | 72 mlir::Value concat_dim = op->getOperand(concat_dim_operand_idx); in ExpandOp() local
117 mlir::Value concat_dim = op->getOperand(concat_dim_operand_idx); in ComputeLayoutForward() local
143 mlir::Value concat_dim = op->getOperand(concat_dim_operand_idx); in ComputeLayoutBackward() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_concat_op.cc | 109 const int32_t concat_dim) { in CalculateInputsDim()
119 OpKernelContext* context, const int32_t concat_dim, in CalculateConcatDims()
168 const int32_t concat_dim = concat_dim_tensor->scalar<int32>()(); in Compute() local
|
| H A D | concat_op.cc | 73 int64_t concat_dim; in Compute() local
249 const Tensor& concat_dim = ctx->input(0); in Compute() local
|
| H A D | concat_op_test.cc | 67 Tensor concat_dim(DT_INT32, TensorShape({})); in ConcatHelper() local
165 Tensor concat_dim(DT_INT32, TensorShape({})); in ConcatManyHelper() local
|
| H A D | sparse_concat_op.cc | 155 const int concat_dim = (concat_dim_attr_ < 0) in Compute() local
|
| H A D | sparse_concat_op_gpu.cu.cc | 41 int64 output_nnz, int rank, int concat_dim, bool need_to_sort, in SparseConcatKernel()
|
| H A D | quantized_concat_op_test.cc | 259 Tensor concat_dim = test::AsScalar<int32>(concat_dimension); in ConcatHelper() local
|
| /aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
| H A D | concat.py | 9 def __init__(self, mode, device, dtype, I1_D1, I1_D2, I2_D1, I2_D2, concat_dim): argument
76 def __init__(self, mode, device, dtype, I1_D1, I1_D2, I2_D1, I2_D2, concat_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_concat_op.cc | 151 const int32 concat_dim = in Compute() local
485 int32 concat_dim = in Compute() local
887 int concat_dim, in FindMklCommonFormat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_simplifier.cc | 68 int64_t concat_dim = concat->concatenate_dimension(); in SliceConcatForwarding() local
|
| H A D | while_loop_concat_code_motion.cc | 161 int64_t concat_dim; member
|
| H A D | dynamic_padder.cc | 1298 const int64_t concat_dim = concat->concatenate_dimension(); in RewriteDynamicConcat() local
|
| H A D | algebraic_simplifier.cc | 1562 const int64_t concat_dim = concat->concatenate_dimension(); in TrySimplifyTautologicalBitcastConvert() local
4704 int64_t concat_dim = concat->concatenate_dimension(); in HandleSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | concat_op.cc | 52 int64_t concat_dim; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | testlib.cc | 358 Node* Concat(Graph* g, Node* concat_dim, gtl::ArraySlice<Node*> tensors) { in Concat()
372 Node* ConcatV2(Graph* g, gtl::ArraySlice<Node*> tensors, Node* concat_dim) { in ConcatV2()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/ |
| H A D | Tensor.h | 408 inline int32_t concat_dim() const { in concat_dim() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | shape_refiner_test.cc | 1091 auto concat_dim = ops::Const(root, 0); in TEST_F() local
1133 auto concat_dim = ops::Const(root, 0); in TEST_F() local
1175 auto concat_dim = ops::Const(root, 0); in TEST_F() local
|
| H A D | shape_refiner.cc | 555 const int concat_dim = in ConstantPartialShape() local
|
| H A D | constant_folding_test.cc | 315 auto concat_dim = ops::Const<int>(s, 0); in TEST_F() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_deepspeech_v0_4_1.cpp | 253 …const DataLayoutDimension concat_dim = (common_params.data_layout == DataLayout::NHWC) ? DataLayou… in add_lstm_cell() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_grad.py | 68 def _CreateDenseMaskAndBegin(sizes, concat_dim): argument
|
| H A D | sparse_ops.py | 283 concat_dim=None, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| H A D | misc.cpp | 546 int64_t concat_dim = std::get<int64_t>(arg_dim); in computeCatWoConditionals() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.cc | 1026 uint64_t concat_dim = split_dim; in ImportInstructionImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | lower_tf.cc | 1625 auto concat_dim = in matchAndRewrite() local
|