Searched refs:sa_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 452 std::vector<InputDesc>* inputs, TensorShape* sa_shape) { in AnalyzeInputs() argument
477 *sa_shape = TensorShape({num_elts}); in AnalyzeInputs()
515 const std::vector<InputDesc>& inputs, const TensorShape& sa_shape) { in ConstructScopedAllocatorNode() argument
523 sa_builder.Attr("shape", sa_shape); in ConstructScopedAllocatorNode()
602 const TensorShape& sa_shape, in BuildSAConcatNode() argument
636 << sa_shape.DebugString(); in BuildSAConcatNode()
641 sac_builder.Attr("shape", sa_shape); in BuildSAConcatNode()
844 TensorShape sa_shape; in Rewrite() local
849 &device_name, &dtype, &input_shapes, &inputs, &sa_shape)); in Rewrite()
856 input_shapes, inputs, sa_shape)); in Rewrite()
[all …]
|