Searched refs:size_array (Results 1 – 4 of 4) sorted by relevance
99 size_array="$OPTARG"132 if [ x"$size_array" = x ]; then135 for size in $size_array ; do
47 const auto& size_array = model->GetArray(op->inputs[2]); in Run() local48 if (!size_array.has_shape()) return ::tensorflow::Status::OK(); in Run()51 op->size = size_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
805 const auto& size_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowSplitVOperator() local807 if (!size_array.has_shape()) { in ProcessTensorFlowSplitVOperator()810 const Shape& size_shape = size_array.shape(); in ProcessTensorFlowSplitVOperator()812 CHECK(size_array.data_type == ArrayDataType::kInt32 || in ProcessTensorFlowSplitVOperator()813 size_array.data_type == ArrayDataType::kInt64) in ProcessTensorFlowSplitVOperator()818 if (size_array.data_type == ArrayDataType::kInt32) { in ProcessTensorFlowSplitVOperator()820 size_array.GetBuffer<ArrayDataType::kInt32>().data) { in ProcessTensorFlowSplitVOperator()824 size_splits_vector = size_array.GetBuffer<ArrayDataType::kInt64>().data; in ProcessTensorFlowSplitVOperator()
37 #### Constructor Tensor<data_type, rank>(size_array)