Searched refs:merge_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_merge_dims_op_test.py | 188 actual = x.merge_dims(outer_axis, inner_axis)
196 actual_with_neg_axis = x.merge_dims(outer_axis - x.shape.rank,
207 actual_with_placeholders = x_with_placeholders.merge_dims(
264 self.evaluate(x.merge_dims(outer_axis, inner_axis))
|
| D | ragged_tensor.py | 1464 def merge_dims(self, outer_axis, inner_axis): member in RaggedTensor
1510 return merge_dims(self, outer_axis, inner_axis)
2866 def merge_dims(value, outer_axis, inner_axis): function
2904 merge_dims(value.values, outer_axis - 1, inner_axis - 1))
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 1075 def merge_dims(self, outer_axis, inner_axis): member in StructuredTensor
1557 return ragged_tensor.merge_dims(value, outer_axis, inner_axis)
1616 return source.merge_dims(outer, inner)
1618 return ragged_tensor.merge_dims(source, outer, inner)
|
| D | structured_tensor_test.py | 1264 result = st.merge_dims(outer_axis, inner_axis)
1270 result = st.merge_dims(0, 1)
1283 merged = struct_3.merge_dims(0, 1)
1291 st.merge_dims(2, 1)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 100 name: "merge_dims"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 100 name: "merge_dims"
|
| /external/tensorflow/ |
| D | RELEASE.md | 6199 * Add `RaggedTensor.merge_dims()`.
|