Searched refs:size_array (Results 1 – 7 of 7) 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()
818 const auto& size_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowSplitVOperator() local820 if (!size_array.has_shape()) { in ProcessTensorFlowSplitVOperator()823 const Shape& size_shape = size_array.shape(); in ProcessTensorFlowSplitVOperator()825 CHECK(size_array.data_type == ArrayDataType::kInt32 || in ProcessTensorFlowSplitVOperator()826 size_array.data_type == ArrayDataType::kInt64) in ProcessTensorFlowSplitVOperator()831 if (size_array.data_type == ArrayDataType::kInt32) { in ProcessTensorFlowSplitVOperator()833 size_array.GetBuffer<ArrayDataType::kInt32>().data) { in ProcessTensorFlowSplitVOperator()837 size_splits_vector = size_array.GetBuffer<ArrayDataType::kInt64>().data; in ProcessTensorFlowSplitVOperator()
85 CPDF_Array* size_array = func_dict->SetNewFor<CPDF_Array>("Size"); in CreateType0FunctionStream() local86 size_array->AddNew<CPDF_Number>(4); in CreateType0FunctionStream()
311 bool points, mali_ptr size_array, in panfrost_emit_primitive_size() argument318 cfg.size_array = size_array; in panfrost_emit_primitive_size()
1110 if (ps.size_array == 0x0) in pandecode_primitive_size()
37 #### Constructor Tensor<data_type, rank>(size_array)