Searched refs:flat_values_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 1362 flat_values_shape = array_ops.shape(rt_flat_values, out_type=out_type)
1368 inner_dimensions = flat_values_shape[1:]
2484 flat_values_shape = tensor_shape.TensorShape([None]).concatenate(
2486 flat_values_spec = tensor_spec.TensorSpec(flat_values_shape, self._dtype)
2842 flat_values_shape = array_ops.shape(op_inputs_flat_values)
2844 flat_values_shape)
|
| D | dynamic_ragged_shape.py | 1076 flat_values_shape = flat_values.shape
1077 if flat_values_shape is None:
1079 first_dim_flat_values = flat_values_shape[0]
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 663 Output outer_shape, flat_values_shape; in BatchGatherGrad() local
671 flat_values_shape = Concat(scope, {{-1}, inner_shape}, /*axis=*/0); in BatchGatherGrad()
674 values = Reshape(scope, values, flat_values_shape); in BatchGatherGrad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 629 flat_values_shape = array_ops.concat([[-1], inner_shape], 0)
634 _IndexedSlicesToTensorNoWarning(values), flat_values_shape)
642 params_grad, array_ops.concat([outer_shape, flat_values_shape], 0))
|