Searched defs:size_splits (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | split_v.cc | 42 const TfLiteTensor* size_splits; member
56 void GetSizeSplitsVector(const TfLiteTensor* size_splits, in GetSizeSplitsVector()
66 const TfLiteTensor* size_splits, in ResizeOutputTensors()
147 auto size_splits = op_context.size_splits; in Prepare() local
|
| D | split_v_test.cc | 41 SplitVOpModel(const TensorData& input, const TensorData& size_splits, in SplitVOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op_test.cc | 60 static Graph* MakeGraph(int split_dim, const std::vector<int64_t>& size_splits, in MakeGraph()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | SplitLayerNode.cpp | 35 SplitLayerNode::SplitLayerNode(unsigned int num_splits, int axis, std::vector<int> size_splits) in SplitLayerNode()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 241 std::vector<Tensor> SplitVGrad(const Tensor& x, const Tensor& size_splits, in SplitVGrad()
297 Tensor size_splits(DT_INT64, {2}); in TEST() local
|
| D | array_ops.cc | 618 const Tensor* size_splits = c->input_tensor(1); in __anon091a71b70b02() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 619 const auto* size_splits = size_splits_tensor.data.i32; in ComputeSplitVUnknownSplitSize() local
1333 const auto* size_splits = size_splits_tensor.data.i32; in TransformSplitVIntoSupportedOps() local
3286 const TfLiteTensor& size_splits = context->tensors[node->inputs->data[1]]; in Validate() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2437 llvm::SmallVector<int64_t, 4> size_splits; in verify() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 3459 auto size_splits = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 4383 std::vector<int32> size_splits(n); in TEST_F() local
|