Searched refs:rshape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_gemm_pad_for_tensor_cores.cc | 34 Shape rshape = rhs->shape(); in PadForTensorCores() local
38 rshape.element_type() != PrimitiveType::F16) { in PadForTensorCores()
56 Shape new_rshape = pad_matrix_dims(rshape); in PadForTensorCores()
59 if (new_lshape == lshape && new_rshape == rshape) { in PadForTensorCores()
63 VLOG(3) << "old shape: " << lshape << " " << rshape << " " << result_shape; in PadForTensorCores()
80 auto r_padding_config = create_padding_config(rshape, new_rshape); in PadForTensorCores()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/examples/ |
| D | LabelImage.java | 117 final long[] rshape = result.shape(); in executeInceptionGraph() local
118 if (result.numDimensions() != 2 || rshape[0] != 1) { in executeInceptionGraph()
122 Arrays.toString(rshape))); in executeInceptionGraph()
124 int nlabels = (int) rshape[1]; in executeInceptionGraph()
|
| /external/python/cpython3/Lib/test/ |
| D | test_buffer.py | 394 rshape = slice_shape(rlst, rslices)
395 if (len(lshape) != len(rshape)):
398 if lshape[i] != rshape[i]:
587 lshape = [0]*ndim; rshape = [0]*ndim
610 rshape[n], lshape[n] = big, small
613 rshape[n], lshape[n] = small, big
616 return lshape, rshape, tuple(lslices), tuple(rslices)
1950 lshape, rshape, lslices, rslices = \
1953 tr = rand_structure(itemsize, True, shape=rshape)
1986 if 0 in lshape or 0 in rshape:
|