Searched defs:shape_spec (Results 1 – 7 of 7) sorted by relevance
112 const tstring& shape_spec = tensor_shapes_and_slices_ptr[i]; in SaveTensors() local208 const tstring& shape_spec = in RestoreTensor() local
121 const string& shape_spec = shape_and_slices_flat(i); in Compute() local
343 StridedSliceShapeSpec shape_spec; in Compute() local
62 const StridedSliceShapeSpec& shape_spec, in EmitDynamicSlice()264 StridedSliceShapeSpec shape_spec; in Compile() local434 StridedSliceShapeSpec shape_spec; in CompileAsDynamicUpdateSlice() local
68 std::string shape_spec = absl::StrJoin(global_shape, " "); in BuildSliceSpecDeviceMap() local
179 StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp()431 StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp()
26 auto shape_spec = [&shape_spec_middle, shape_spec_end]( in TestGradAndIndicesErrorHandling() local