Searched defs:size_splits (Results 1 – 7 of 7) 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()
144 auto size_splits = op_context.size_splits; in Prepare() local
|
| D | split_v_test.cc | 35 SplitVOpModel(const TensorData& input, const TensorData& size_splits, in SplitVOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op_test.cc | 57 static Graph* MakeGraph(int split_dim, const std::vector<int64>& size_splits, in MakeGraph()
|
| /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 | 617 const Tensor* size_splits = c->input_tensor(1); in __anon42d741190b02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1945 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
|