HomeSort by relevance Sort by last modified time
    Searched full:init_shape (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment.cc 695 auto init_shape = while_inst->operand(0)->shape(); local
698 init_shape));
700 init_shape));
702 body_computation_layout.result_layout().MatchesLayoutInShape(init_shape));
704 LayoutUtil::LayoutsInShapesEqual(init_shape, while_inst->shape()));
    [all...]
  /external/tensorflow/tensorflow/python/ops/
variable_scope.py 638 init_shape = None
640 init_shape = var_shape
643 init_shape = var_shape
648 init_shape = None
652 init_shape = None
657 shape=init_shape,
    [all...]
  /external/tensorflow/tensorflow/contrib/metrics/python/ops/
metric_ops.py     [all...]

Completed in 104 milliseconds