Searched refs:split_dim_tensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | split_op.cc | 50 const Tensor& split_dim_tensor = context->input(0); in ComputeEasyCases() local
52 context, split_dim_tensor.shape().dims() == 0, in ComputeEasyCases()
54 split_dim_tensor.shape().dims())); in ComputeEasyCases()
55 const int32_t split_dim_orig = split_dim_tensor.flat<int32>()(0); in ComputeEasyCases()
|
| H A D | split_op_test.cc | 36 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); in MakeGraph() local
39 .Input(test::graph::Constant(g, split_dim_tensor)) in MakeGraph()
|
| H A D | split_v_op_test.cc | 66 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); in MakeGraph() local
72 .Input(test::graph::Constant(g, split_dim_tensor)) in MakeGraph()
|
| H A D | split_v_op.cc | 59 const Tensor& split_dim_tensor = context->input(2); in ComputeEasyCases() local
61 OP_REQUIRES(context, split_dim_tensor.NumElements() == 1, in ComputeEasyCases()
65 const int32_t split_dim_orig = split_dim_tensor.flat<int32>()(0); in ComputeEasyCases()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 2353 const TfLiteTensor& split_dim_tensor = tensors[split_dim_idx]; in VisitSplitNode() local
2354 TF_LITE_ENSURE_STATUS(CheckTensorType(logging_context, split_dim_tensor, in VisitSplitNode()
2358 logging_context, split_dim_tensor, split_dim_idx, node_index)); in VisitSplitNode()
2367 int32_t split_dim = GetTensorData<int32_t>(&split_dim_tensor)[0]; in VisitSplitNode()
|