Searched defs:shape_spec (Results 1 – 8 of 8) sorted by relevance
62 const StridedSliceShapeSpec& shape_spec, in EmitDynamicSlice()264 StridedSliceShapeSpec shape_spec; in Compile() local434 StridedSliceShapeSpec shape_spec; in CompileAsDynamicUpdateSlice() local
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
179 StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp()431 StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp()
68 std::string shape_spec = absl::StrJoin(global_shape, " "); in BuildSliceSpecDeviceMap() local
26 auto shape_spec = [&shape_spec_middle, shape_spec_end]( in TestGradAndIndicesErrorHandling() local
152 const tensorflow::tstring& shape_spec = tensor_shapes_and_slices_ptr[i]; in SaveTensors() local