Searched refs:step_tensor (Results 1 – 3 of 3) sorted by relevance
36 const auto& step_tensor = context->tensors[inputs->data[3]]; in PopulateSubGraph() local42 graph_builder_->AddConstNodeWithData(inputs->data[3], step_tensor); in PopulateSubGraph()
442 const auto& step_tensor = context->tensors[node->inputs->data[3]]; in IsNodeSupportedByHexagon() local444 !IsConstantTensor(&ends_tensor) || !IsConstantTensor(&step_tensor)) in IsNodeSupportedByHexagon()
1023 Tensor step_tensor = CreateTensor<int64_t>(TensorShape({}), {step_}); in GetInputTensors() local1024 return {start_tensor, stop_tensor, step_tensor}; in GetInputTensors()