| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_concat_op.cc | 109 const int32 concat_dim) { in CalculateInputsDim()
119 OpKernelContext* context, const int32 concat_dim, in CalculateConcatDims()
168 const int32 concat_dim = concat_dim_tensor->scalar<int32>()(); in Compute() local
|
| D | concat_op.cc | 65 int64 concat_dim; in Compute() local
281 const Tensor& concat_dim = ctx->input(0); in Compute() local
|
| D | mkl_concat_op.cc | 75 const int32 concat_dim = in Compute() local
175 int32 concat_dim = in Compute() local
498 int concat_dim, bool* is_reorder_needed, in FindMklCommonFormat()
|
| D | concat_op_test.cc | 47 Tensor concat_dim(DT_INT32, TensorShape({})); in ConcatHelper() local
103 Tensor concat_dim(DT_INT32, TensorShape({})); in ConcatManyHelper() local
|
| D | sparse_concat_op.cc | 78 const int concat_dim = (concat_dim_attr_ < 0) in Compute() local
|
| D | quantized_concat_op_test.cc | 260 Tensor concat_dim = test::AsScalar<int32>(concat_dimension); in ConcatHelper() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | concat_op.cc | 52 int64 concat_dim; in Compile() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 670 auto concat_dim = node_map.GetNode("concat-3-LayoutOptimizer"); in TEST_F() local
691 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
713 auto concat_dim = in TEST_F() local
736 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
757 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
778 auto concat_dim = node_map.GetNode("concat-2-LayoutOptimizer"); in TEST_F() local
|
| D | constant_folding_test.cc | 1156 Tensor concat_dim(DT_INT32, TensorShape({})); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 357 Node* Concat(Graph* g, Node* concat_dim, gtl::ArraySlice<Node*> tensors) { in Concat()
371 Node* ConcatV2(Graph* g, gtl::ArraySlice<Node*> tensors, Node* concat_dim) { in ConcatV2()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 516 const int concat_dim = in ConstantPartialShape() local
|
| D | shape_refiner_test.cc | 1087 auto concat_dim = ops::Const(root, 0); in TEST_F() local
1129 auto concat_dim = ops::Const(root, 0); in TEST_F() local
1171 auto concat_dim = ops::Const(root, 0); in TEST_F() 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 | 66 def _CreateDenseMaskAndBegin(sizes, concat_dim): argument
|
| D | sparse_ops.py | 200 concat_dim=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 457 const int64 concat_dim = concatenate->dimensions()[0]; in HandleConcatenate() local
|
| D | elemental_ir_emitter.cc | 1709 const int64 concat_dim = hlo->dimensions(0); in EmitElementalConcatenate() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1250 const int32 concat_dim = concat_dim_t->scalar<int32>()(); in ConcatShapeHelper() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1435 int concat_dim = in TEST_F() local
1456 int concat_dim = in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 598 inline void Concatenation(int concat_dim, const Scalar* const* input_data, in Concatenation()
618 inline void Concatenation(int concat_dim, const uint8* const* input_data, in Concatenation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2405 int64 concat_dim = concatenate->dimensions(0); in EmitFastConcatenate() local
|