Searched refs:slice_inputs (Results 1 – 2 of 2) sorted by relevance
103 SliceInputs slice_inputs; in GetSliceInputs() local104 slice_inputs.input = in GetSliceInputs()106 slice_inputs.begin = in GetSliceInputs()108 slice_inputs.size = in GetSliceInputs()112 TryToGetTensorFromConstOp(slice_inputs.size.node())); in GetSliceInputs()121 slice_inputs.size_as_vector = IntTensorAsVector(*tf_slice_size); in GetSliceInputs()122 return {slice_inputs}; in GetSliceInputs()135 const SliceInputs& slice_inputs) { in MakeSliceIndexAndSizeInt64() argument137 result.input = slice_inputs.input; in MakeSliceIndexAndSizeInt64()138 result.begin = MakeInt64(host_scope, "begin", slice_inputs.begin); in MakeSliceIndexAndSizeInt64()[all …]
356 dataset = self.slice_inputs(indices_dataset, inputs)365 def slice_inputs(self, indices_dataset, inputs): member in TensorLikeDataAdapter470 def slice_inputs(self, indices_dataset, inputs): member in GenericArrayLikeDataAdapter