Searched refs:batched_shape (Results 1 – 3 of 3) sorted by relevance
379 const BHWC batched_shape = in ConvertToPHWC4() local381 std::vector<float> transposed(GetElementsSizeForPHWC4(batched_shape)); in ConvertToPHWC4()382 ConvertToPHWC4(tensor.data, batched_shape, in ConvertToPHWC4()
504 batched_shape = [batch_size, num_states, num_states]505 label_to_label = array_ops.scatter_nd(indices, values, batched_shape)
1150 batched_shape = tensor_shape.TensorShape([self._maybe_iters1153 tensor_spec.TensorSpec(shape=batched_shape, dtype=arg.dtype))