Searched refs:host_scope (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 126 Output MakeInt64(const Scope& host_scope, absl::string_view name, in MakeInt64() argument
130 : ops::Cast(host_scope.WithOpName(name, "_s64"), x, DT_INT64); in MakeInt64()
134 SliceInputs MakeSliceIndexAndSizeInt64(const Scope& host_scope, in MakeSliceIndexAndSizeInt64() argument
138 result.begin = MakeInt64(host_scope, "begin", slice_inputs.begin); in MakeSliceIndexAndSizeInt64()
139 result.size = MakeInt64(host_scope, "size", slice_inputs.size); in MakeSliceIndexAndSizeInt64()
172 Status ComputeSliceSize(const Scope& host_scope, in ComputeSliceSize() argument
191 ops::Shape(host_scope.WithOpName("input_shape"), slice_inputs.input, in ComputeSliceSize()
194 ConstantCache constant_pool(host_scope, control_deps); in ComputeSliceSize()
207 host_scope.WithOpName("begin_", i), slice_inputs.begin, in ComputeSliceSize()
211 host_scope.WithOpName("input_shape_", i), input_shape, in ComputeSliceSize()
[all …]
|