| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 486 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); local
493 HloInstruction::CreateParameter(0, full_shape, "slice_from")),
499 full_shape,
501 HloInstruction::CreateParameter(2, full_shape, "to_update")),
517 Shape full_shape = ShapeUtil::MakeShape(F32, {4, 100, 10, 100, 50}); local
552 full_shape,
554 HloInstruction::CreateParameter(4, full_shape, "param4")),
|
| /external/tensorflow/tensorflow/python/ops/ |
| variable_scope.py | 55 def __init__(self, full_shape, var_offset):
59 full_shape: Tuple or list of `int` indicating the full combined shape
65 TypeError: If `full_shape` or `var_offset` is not a sequence.
66 ValueError: If `full_shape` or `var_offset` differ in length. If
67 `var_offset` exceeds `full_shape` in any dimension.
69 if not isinstance(full_shape, collections_lib.Sequence) or isinstance(
70 full_shape, six.string_types):
72 "`full_shape` must be a sequence (like tuple or list) instead of " +
73 type(full_shape).__name__)
81 if len(var_offset) != len(full_shape)
100 def full_shape(self): member in class:_PartitionInfo
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | 2406 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); local
[all...] |
