| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| D | concat_op.cc | 74 int64_t concat_dim; in Compute() local
250 const Tensor& concat_dim = ctx->input(0); in Compute() local
|
| 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
|
| D | sparse_concat_op.cc | 155 const int concat_dim = (concat_dim_attr_ < 0) in Compute() local
|
| D | quantized_concat_op_test.cc | 259 Tensor concat_dim = test::AsScalar<int32>(concat_dimension); in ConcatHelper() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_simplifier.cc | 68 int64_t concat_dim = concat->concatenate_dimension(); in SliceConcatForwarding() local
|
| D | while_loop_concat_code_motion.cc | 160 int64 concat_dim; member
|
| D | dynamic_padder.cc | 1248 const int64_t concat_dim = concat->concatenate_dimension(); in RewriteDynamicConcat() local
|
| D | hlo_evaluator.cc | 529 const int64_t concat_dim = concatenate->dimensions()[0]; in HandleConcatenate() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 52 int64_t concat_dim; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 148 const int32 concat_dim = in Compute() local
475 int32 concat_dim = in Compute() local
877 int concat_dim, in FindMklCommonFormat()
|
| /external/ComputeLibrary/examples/ |
| D | graph_deepspeech_v0_4_1.cpp | 251 …const DataLayoutDimension concat_dim = (common_params.data_layout == DataLayout::NHWC) ? DataLayou… in add_lstm_cell() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 690 auto concat_dim = node_map.GetNode("concat-3-LayoutOptimizer"); in TEST_F() local
711 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
733 auto concat_dim = in TEST_F() local
756 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
777 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
798 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 1072 auto concat_dim = ops::Const(root, 0); in TEST_F() local
1114 auto concat_dim = ops::Const(root, 0); in TEST_F() local
1156 auto concat_dim = ops::Const(root, 0); in TEST_F() local
|
| D | shape_refiner.cc | 551 const int concat_dim = in ConstantPartialShape() local
|
| D | constant_folding_test.cc | 315 auto concat_dim = ops::Const<int>(s, 0); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 72 def _CreateDenseMaskAndBegin(sizes, concat_dim): argument
|
| D | sparse_ops.py | 287 concat_dim=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1633 auto concat_dim = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1964 int64_t concat_dim; in ConcatShapeHelper() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 648 inline void Concatenation(int concat_dim, const Scalar* const* input_data, in Concatenation()
668 inline void Concatenation(int concat_dim, const uint8* const* input_data, in Concatenation()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1436 int concat_dim = in TEST_F() local
1457 int concat_dim = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1407 int32_t concat_dim = (*concat_dim_attr.getValues<int32_t>().begin()); in fold() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2489 int64_t concat_dim = concatenate->dimensions(0); in EmitFastConcatenate() local
|