Searched refs:orig_input_tensor_shape_flat (Results 1 – 1 of 1) sorted by relevance
251 auto orig_input_tensor_shape_flat = orig_input_tensor_shape.flat<int64>(); in Compute() local253 const int64 in_batch = orig_input_tensor_shape_flat(0); in Compute()254 const int64 in_rows = orig_input_tensor_shape_flat(1); in Compute()255 const int64 in_cols = orig_input_tensor_shape_flat(2); in Compute()256 const int64 in_depth = orig_input_tensor_shape_flat(3); in Compute()262 in_shape.AddDim(orig_input_tensor_shape_flat(i)); in Compute()