Searched refs:shape_internal (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | array_ops.py | 313 return shape_internal(input, name, optimize=True, out_type=out_type) 316 def shape_internal(input, name=None, optimize=True, out_type=dtypes.int32): function 1914 shape_internal(tensor, optimize=optimize), dtype=dtype, name=name) 1929 shape_internal(tensor, optimize=optimize), dtype=dtype, name=name) 2001 ones_shape = shape_internal(tensor, optimize=optimize) 3620 sequence_size = shape_internal(sorted_sequence)[-1] 3621 values_size = shape_internal(values)[-1] 3632 return reshape(output, shape_internal(values))
|
D | control_flow_ops.py | 1172 shape = array_ops.shape_internal(val, optimize=False) 1237 zeros_shape = array_ops.shape_internal(val, optimize=False) 1243 zeros_shape = array_ops.shape_internal(val, optimize=False) 1289 grad_shape = array_ops.shape_internal(enter_grad, optimize=False) 1345 zeros_shape = array_ops.shape_internal(switch_val, optimize=False) 2156 return array_ops.shape_internal(t, optimize=False) 2698 zeros_shape = array_ops.shape_internal(value, optimize=False) 2711 zeros_shape = array_ops.shape_internal(value, optimize=False) 2783 array_ops.shape_internal(
|
D | math_ops.py | 2724 x_shape = array_ops.shape_internal(x, optimize=optimize) 3213 segment_ids_shape = array_ops.shape_internal(segment_ids)
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | unary_ops_test.py | 848 shape_op = lambda x: array_ops.shape_internal(x, optimize=False)
|
/external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
D | metric_ops.py | 3655 array_size = array_ops.shape_internal(array, optimize=False)[0]
|